package com.github.wxhnyfy;

import java.util.List;
import java.util.Map;

/**
 * Excel数据
 *
 * @author chenwc
 */
public class ExcelData {
    /**
     * 文件名
     */
    private String fileName;
    /**
     * Excel数据
     */
    private List<SheetData> sheetDataList;

    /**
     * Sheet页名，key为sheet页索引，value为sheet页名
     */
    private Map<Integer, String> sheetNameMap;

    /**
     * Sheet页数
     */
    private int sheetSize;

    public String getFileName() {
        return fileName;
    }

    public void setFileName(String fileName) {
        this.fileName = fileName;
    }

    public List<SheetData> getSheetDataList() {
        return sheetDataList;
    }

    public void setSheetDataList(List<SheetData> sheetDataList) {
        this.sheetDataList = sheetDataList;
    }

    public int getSheetSize() {
        return sheetSize;
    }

    public void setSheetSize(int sheetSize) {
        this.sheetSize = sheetSize;
    }

    /**
     * Sheet页名，key为sheet页索引，value为sheet页名
     */
    public void setSheetNameMap(Map<Integer, String> sheetNameMap) {
        this.sheetNameMap = sheetNameMap;
    }

    /**
     * 获取指定sheet页数据
     *
     * @param sheetIndex sheet页索引
     * @return sheet页数据
     */
    public SheetData getSheetData(int sheetIndex) {
        return sheetDataList.get(sheetIndex);
    }

    /**
     * 获取指定sheet页数据
     *
     * @param sheetName sheet页名
     * @return sheet页数据
     */
    public SheetData getSheetData(String sheetName) {
        for (int i = 0; i < sheetSize; i++) {
            if (sheetName.equals(sheetNameMap.get(i))) {
                return sheetDataList.get(i);
            }
        }
        return null;
    }

    /**
     * 获取指定sheet页名
     *
     * @param sheetIndex sheet页索引
     * @return sheet页名
     */
    public String getSheetName(int sheetIndex) {
        return sheetNameMap.get(sheetIndex);
    }
}
